home *** CD-ROM | disk | FTP | other *** search
/ Programming an RTS Game with Direct3D / Programming an RTS Game with Direct3D.iso / Examples / Chapter 5 / Example 5.9 / mouse.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  2006-07-12  |  7.0 KB  |  275 lines
  1. #include "mouse.h"
  2.  
  3. //////////////////////////////////////////////////////////////////////////
  4. //                        RAY                                                //
  5. //////////////////////////////////////////////////////////////////////////
  6.  
  7. RAY::RAY()
  8. {
  9.     org = dir = D3DXVECTOR3(0.0f, 0.0f, 0.0f);
  10. }
  11.  
  12. RAY::RAY(D3DXVECTOR3 o, D3DXVECTOR3 d)
  13. {
  14.     org = o;
  15.     dir = d;
  16. }
  17.  
  18. float RAY::Intersect(MESHINSTANCE iMesh)
  19. {
  20.     if(iMesh.m_pMesh == NULL)return -1.0f;
  21.     return Intersect(iMesh.m_pMesh->m_pMesh);
  22. }
  23.  
  24. float RAY::Intersect(BBOX bBox)
  25. {
  26.     if(D3DXBoxBoundProbe(&bBox.min, &bBox.max, &org, &dir))
  27.         return D3DXVec3Length(&(((bBox.min + bBox.max) / 2.0f) - org));
  28.     else return -1.0f;
  29. }
  30.  
  31. float RAY::Intersect(BSPHERE bSphere)
  32. {
  33.     if(D3DXSphereBoundProbe(&bSphere.center, bSphere.radius, &org, &dir))
  34.         return D3DXVec3Length(&(bSphere.center - org));
  35.     else return -1.0f;
  36. }
  37.  
  38. float RAY::Intersect(ID3DXMesh* mesh)
  39. {
  40.     if(mesh == NULL)return -1.0f;
  41.  
  42.     // Collect only the closest intersection
  43.     BOOL hit;
  44.     DWORD dwFace;
  45.     float hitU, hitV, dist;
  46.     D3DXIntersect(mesh, &org, &dir, &hit, &dwFace, &hitU, &hitV, &dist, NULL, NULL);
  47.  
  48.     if(hit) return dist;
  49.     else return -1.0f;
  50. }
  51.  
  52. //////////////////////////////////////////////////////////////////////////
  53. //                        MOUSE                                            //
  54. //////////////////////////////////////////////////////////////////////////
  55.  
  56. MOUSE::MOUSE()
  57. {
  58.     m_type = 0;
  59.     m_pMouseTexture = NULL;
  60.     m_pMouseDevice = NULL;
  61.     m_speed = 1.5f;
  62. }
  63.  
  64. MOUSE::~MOUSE()
  65. {
  66.     if(m_pMouseDevice != NULL)
  67.         m_pMouseDevice->Release();
  68.  
  69.     if(m_pMouseTexture != NULL)
  70.         m_pMouseTexture->Release();
  71. }
  72.  
  73. void MOUSE::InitMouse(IDirect3DDevice9* Dev, HWND wnd)
  74. {
  75.     try
  76.     {
  77.         m_pDevice = Dev;
  78.  
  79.         //Load texture and init sprite
  80.         D3DXCreateTextureFromFile(m_pDevice, "cursor/cursor.dds", &m_pMouseTexture);
  81.         D3DXCreateSprite(m_pDevice, &m_pSprite);
  82.  
  83.         //Get directinput object
  84.         LPDIRECTINPUT8 directInput = NULL;
  85.  
  86.         DirectInput8Create(GetModuleHandle(NULL),    // Retrieves the application handle
  87.                            0x0800,                    // v.8.0    
  88.                            IID_IDirectInput8,        // Interface ID
  89.                            (VOID**)&directInput,    // Our DirectInput Object
  90.                            NULL);
  91.  
  92.         //Acquire Default System mouse
  93.         directInput->CreateDevice(GUID_SysMouse, &m_pMouseDevice, NULL);        
  94.         m_pMouseDevice->SetDataFormat(&c_dfDIMouse);
  95.         m_pMouseDevice->SetCooperativeLevel(wnd, DISCL_EXCLUSIVE | DISCL_FOREGROUND);
  96.         m_pMouseDevice->Acquire();            
  97.  
  98.         //Get m_viewport size and init mouse location
  99.         D3DVIEWPORT9 v;
  100.         m_pDevice->GetViewport(&v);
  101.         m_viewport.left = v.X;
  102.         m_viewport.top = v.Y;
  103.         m_viewport.right = v.X + v.Width;
  104.         m_viewport.bottom = v.Y + v.Height;
  105.  
  106.         x = v.X + v.Width / 2.0f;
  107.         y = v.Y + v.Height / 2.0f;
  108.  
  109.         //Release Direct Input Object
  110.         directInput->Release();
  111.  
  112.         //Create Mouse Sphere
  113.         D3DXCreateSphere(m_pDevice, 0.2f, 5, 5, &m_pSphereMesh, NULL);
  114.  
  115.         //Create Ball Material
  116.         m_ballMtrl.Diffuse = D3DXCOLOR(1.0f, 1.0f, 0.0f, 1.0f);
  117.         m_ballMtrl.Specular = m_ballMtrl.Ambient = m_ballMtrl.Emissive = D3DXCOLOR(0.0f, 0.0f, 0.0f, 1.0f);
  118.     }
  119.     catch(...)
  120.     {
  121.         debug.Print("Error in MOUSE::InitMouse()");
  122.     }    
  123. }
  124.  
  125. bool MOUSE::ClickLeft()
  126. { 
  127.     return m_mouseState.rgbButtons[0];
  128. }
  129.  
  130. bool MOUSE::ClickRight()
  131. {
  132.     return m_mouseState.rgbButtons[1];
  133. }
  134.  
  135. bool MOUSE::WheelUp()
  136. {
  137.     return m_mouseState.lZ > 0.0f;
  138. }
  139.  
  140. bool MOUSE::WheelDown()
  141. {
  142.     return m_mouseState.lZ < 0.0f;
  143. }
  144.  
  145. bool MOUSE::Over(RECT dest)
  146. {
  147.     if(x < dest.left || x > dest.right)return false;
  148.     if(y < dest.top || y > dest.bottom)return false;
  149.     return true;
  150. }
  151.  
  152. bool MOUSE::PressInRect(RECT dest)
  153. {
  154.     return ClickLeft() && Over(dest);
  155. }
  156.  
  157. void MOUSE::Update(TERRAIN &terrain)
  158. {
  159.     //Retrieve mouse state
  160.     ZeroMemory(&m_mouseState, sizeof(DIMOUSESTATE));
  161.     m_pMouseDevice->GetDeviceState(sizeof(DIMOUSESTATE), &m_mouseState);
  162.  
  163.     //Update pointer
  164.     x += m_mouseState.lX * m_speed;
  165.     y += m_mouseState.lY * m_speed;
  166.  
  167.     //Keep mouse pointer within window
  168.     if(x < m_viewport.left)    x = m_viewport.left;
  169.     if(y < m_viewport.top)    y = m_viewport.top;
  170.     if(x > m_viewport.right)    x = m_viewport.right;
  171.     if(y > m_viewport.bottom)    y = m_viewport.bottom;
  172.  
  173.     CalculateMappos(terrain);
  174. }
  175.  
  176. void MOUSE::Paint()
  177. {    
  178.     //Draw ball
  179.     D3DXMATRIX world;
  180.     D3DXMatrixTranslation(&world, m_ballPos.x, m_ballPos.y, m_ballPos.z);
  181.     m_pDevice->SetTransform(D3DTS_WORLD, &world);
  182.     m_pDevice->SetMaterial(&m_ballMtrl);
  183.     m_pDevice->SetTexture(0, NULL);
  184.     m_pSphereMesh->DrawSubset(0);
  185.  
  186.     if(m_pMouseTexture != NULL)
  187.     {
  188.         RECT src[] = {{0,0,20,20}, {0,20,20,40}, {20,20,40,40}, {0,40,20,60}, {20,40,40,60}};
  189.  
  190.         m_pSprite->Begin(D3DXSPRITE_ALPHABLEND);
  191.         m_pSprite->Draw(m_pMouseTexture, &src[m_type], NULL, &D3DXVECTOR3(x, y, 0.0f), 0xffffffff);
  192.         m_pSprite->End();
  193.     }    
  194. }
  195.  
  196. RAY MOUSE::GetRay()
  197. {
  198.     try
  199.     {
  200.         D3DXMATRIX projectionMatrix, viewMatrix, worldViewInverse, worldMatrix;
  201.         
  202.         m_pDevice->GetTransform(D3DTS_PROJECTION, &projectionMatrix);
  203.         m_pDevice->GetTransform(D3DTS_VIEW, &viewMatrix);
  204.         m_pDevice->GetTransform(D3DTS_WORLD, &worldMatrix);
  205.  
  206.         float width = m_viewport.right - m_viewport.left;
  207.         float height = m_viewport.bottom - m_viewport.top;
  208.  
  209.         float angle_x = (((2.0f * x) / width) - 1.0f) / projectionMatrix(0,0);
  210.         float angle_y = (((-2.0f * y) / height) + 1.0f) / projectionMatrix(1,1); 
  211.         
  212.         RAY ray;
  213.         ray.org = D3DXVECTOR3(0.0f, 0.0f, 0.0f);
  214.         ray.dir = D3DXVECTOR3(angle_x, angle_y, 1.0f);
  215.  
  216.         D3DXMATRIX m = worldMatrix * viewMatrix;
  217.         D3DXMatrixInverse(&worldViewInverse, 0, &m);
  218.         D3DXVec3TransformCoord(&ray.org, &ray.org, &worldViewInverse);
  219.         D3DXVec3TransformNormal(&ray.dir, &ray.dir, &worldViewInverse);
  220.         D3DXVec3Normalize(&ray.dir, &ray.dir);
  221.  
  222.         return ray;
  223.     }
  224.     catch(...)
  225.     {
  226.         debug.Print("Error in MOUSE::GetRay()");
  227.     }
  228. }
  229.  
  230. void MOUSE::CalculateMappos(TERRAIN &terrain)
  231. {
  232.     //Get Mouse Ray
  233.     D3DXMATRIX world;
  234.     D3DXMatrixIdentity(&world);
  235.     m_pDevice->SetTransform(D3DTS_WORLD, &world);
  236.     RAY mRay = GetRay();
  237.  
  238.     float minDistance = 10000.0f;
  239.     for(int i=0;i<terrain.m_patches.size();i++)
  240.     {
  241.         if(mRay.Intersect(terrain.m_patches[i]->m_BBox) > 0.0f)
  242.         {
  243.             // Collect only the closest intersection
  244.             BOOL hit;
  245.             DWORD dwFace;
  246.             float hitU, hitV, dist;
  247.             D3DXIntersect(terrain.m_patches[i]->m_pMesh, &mRay.org, &mRay.dir, &hit, &dwFace, &hitU, &hitV, &dist, NULL, NULL);
  248.  
  249.             if(hit && dist < minDistance)
  250.             {
  251.                 minDistance = dist;
  252.                 int tiles = dwFace / 2;        //Two faces to each map tile
  253.                 int tilesPerRow = terrain.m_patches[i]->m_mapRect.right - terrain.m_patches[i]->m_mapRect.left;
  254.                 int y = tiles / tilesPerRow, x = tiles - y * tilesPerRow;
  255.  
  256.                 if(dwFace % 2 == 0)        //Hit upper left face
  257.                 {
  258.                     if(hitU > 0.5f)x++;
  259.                     else if(hitV > 0.5f)y++;
  260.                 }
  261.                 else                    //Hit lower right face
  262.                 {
  263.                     if(hitU + hitV < 0.5f)y++;
  264.                     else if(hitU > 0.5f)x++;
  265.                     else {x++;y++;}
  266.                 }
  267.  
  268.                 //Set mouse map position
  269.                 m_mappos.Set(terrain.m_patches[i]->m_mapRect.left + x, terrain.m_patches[i]->m_mapRect.top + y);
  270.                 m_ballPos = terrain.GetWorldPos(m_mappos);
  271.                 m_uv = D3DXVECTOR2(hitU, hitV);
  272.             }            
  273.         }
  274.     }    
  275. }